/*
 * Licensed to the 南京汇海交通科技有限公司
 *
 * Copyright (c) 2025.
 * All rights reserved.
 *
 * 本软件属南京汇海交通科技有限公司所有，在未获得南京汇海交通科技有限公司正式授权情况下，任何企业和个人，不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。
 */
package com.chframework.mybatis.unid.plus;

import cn.hutool.core.util.IdUtil;
import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
import com.chframework.common.unid.AppGenKey;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.lang.NonNull;
import org.springframework.stereotype.Component;

/**
 * mybatis-plus 分布式全局唯一ID生成策略
 */
@Component
public final class DistributedGlobalIdGenerator implements IdentifierGenerator {

    private static AppGenKey genKey;

    /**
     * 设置 创建分布式全局ID 的对象
     */
    @Autowired
    public void setGenKey(@NonNull AppGenKey genKey) {
        DistributedGlobalIdGenerator.setSyncGenKey(genKey);
    }

    private static synchronized void setSyncGenKey(AppGenKey genKey) {
        DistributedGlobalIdGenerator.genKey = genKey;
    }

    @Override
    public Number nextId(Object entity) {
        return genKey.getUniqueKey();
    }

    @Override
    public String nextUUID(Object entity) {
        return IdUtil.simpleUUID();
    }
}
